import torch
model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet152', pretrained=True)


# from transformers import AutoImageProcessor, ResNetForImageClassification
# import torch
# from datasets import load_dataset

# dataset = load_dataset("huggingface/cats-image")
# image = dataset["test"]["image"][0]

# processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
# model = ResNetForImageClassification.from_pretrained("microsoft/resnet-50")